#include "model.h"
#include <torch/script.h>

namespace model {
    torch::jit::Module load_model(const std::string& model_path) {
        try {
            return torch::jit::load(model_path);
        } catch (const c10::Error& e) {
            std::cerr << "Error loading model: " << e.what() << std::endl;
            throw;
        }
    }
} 